import tool.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

/**
 * 在二叉树中，根节点位于深度 0 处，每个深度为 k 的节点的子节点位于深度 k+1 处。
 *
 * 如果二叉树的两个节点深度相同，但 父节点不同 ，则它们是一对堂兄弟节点。
 *
 * 我们给出了具有唯一值的二叉树的根节点 root ，以及树中两个不同节点的值 x 和 y 。
 *
 * 只有与值 x 和 y 对应的节点是堂兄弟节点时，才返回 true 。否则，返回 false。
 *
 * 示例 1：
 * 输入：root = [1,2,3,4], x = 4, y = 3
 * 输出：false
 *
 * 示例 2：
 * 输入：root = [1,2,3,null,4,null,5], x = 5, y = 4
 * 输出：true
 *
 * 示例 3：
 * 输入：root = [1,2,3,null,4], x = 2, y = 3
 * 输出：false
 *
 * 提示：
 * 二叉树的节点数介于2 到100之间。
 * 每个节点的值都是唯一的、范围为1 到100的整数。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/cousins-in-binary-tree
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Q00993s {
    public boolean isCousins(TreeNode root, int x, int y) {
        if (root.val == x || root.val == y) return false;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        int size;
        while ((size = queue.size()) > 0) {
            boolean findX = false, findY = false; // 标记在当前行是否发现x,y
            while (size-- > 0) {
                int l = 0, r = 0; // 标记同父节点的左右子树的值
                root = queue.poll();
                if (root.left != null) {
                    l = root.left.val;
                    queue.offer(root.left);
                }
                if (root.right != null) {
                    r = root.right.val;
                    queue.offer(root.right);
                }
                if ((x == l && y == r) || (x == r && y == l)) return false; // 同一父节点的左右子树，不符
                if (x == l || x == r) findX = true; // 发现x
                else if (y == l || y == r) findY = true; // 发现y
            }
            if (findX && findY) return true; // 在当前层发现是堂兄弟
            if (findX || findY) return false; // 只发现一个，说明肯定不是堂兄弟
        }
        return true;
    }
}
